import time
import streamlit as st

from tongyi_breaking import call_agent_breaking
from tongyi_diabetes import call_agent_diabetes
from tongyi_sensor_qa import call_agent_sensor


def main():
    # 设置页面标题
    st.title("阿开的问答工具箱")

    # 创建标签页
    tab1, tab2, tab3 = st.tabs(["霹雳舞大全","糖尿病问答", "EasySense问答"])

    with tab1:
        # 问答助手标签页
        st.header("霹雳舞大全")

        # 显示文本输入框，让用户输入问题
        user_question_1 = st.text_input("请输入您的问题:",key="input_breaking")

        # 当用户点击“提交”按钮后执行的操作
        if st.button('提交',key="btn_breaking"):
            if user_question_1:
                with st.spinner('正在获取答案...'):  # 显示加载提示
                    # 调用函数获取答案
                    response = call_agent_breaking(user_question_1)
                st.write(f" {response}")
            else:
                st.warning("请先输入一个问题！")

    with tab2:
        # 问答助手标签页
        st.header("糖尿病问答")

        # 显示文本输入框，让用户输入问题
        user_question_2 = st.text_input("请输入您的问题:",key="input_diabetes")

        # 当用户点击“提交”按钮后执行的操作
        if st.button('提交',key="btn_diabetes"):
            if user_question_2:
                with st.spinner('正在获取答案...'):  # 显示加载提示
                    # 调用函数获取答案
                    response = call_agent_diabetes(user_question_2)
                st.write(f" {response}")
            else:
                st.warning("请先输入一个问题！")

    with tab3:
        # 问答助手标签页
        st.header("EasySense问答")

        # 显示文本输入框，让用户输入问题
        query_keyword  = st.text_input("请输入您的问题:",key="input_sensor")

        # 当用户点击“提交”按钮后执行的操作
        if st.button('提交',key="btn_sensor"):
            if query_keyword:
                with st.spinner('正在获取答案...'):  # 显示加载提示
                    # 调用函数获取答案
                    response = call_agent_sensor(query_keyword )
                st.write(f" {response}")
            else:
                st.warning("请先输入一个问题！")

    # 在页面底部添加知识库引用
    st.markdown(
        """
        <style>
        .footer {
            font-size: 10px;
            color: #777;  /* 灰色 */
            text-align: center;
            margin-top: 50px;
        }
        .book-title {
            font-weight: bold;
        }
        .book-info {
            margin-bottom: 10px;
        }
        </style>
        <div class="footer">
        本应用基于以下书籍提供糖尿病相关信息：<br><br>
        <div class="book-info">
            <span class="book-title">《糖尿病病人最关心的365个问题 (常见病病人最关心的问题丛书)》</span><br>
            作者: 李延兵 & 关向东<br>
            出版社: 湖北科学技术出版社<br>
            出版日期: 2015
        </div>
        <div class="book-info">
            <span class="book-title">《糖尿病病人最关心的300个问题》</span><br>
            作者: 关向东<br>
            出版社: 湖北科学技术出版社<br>
            出版日期: 2010
        </div>
        </div>
        """,
        unsafe_allow_html=True
    )

if __name__ == "__main__":
    main()